from langchain_ollama import ChatOllama
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser


# 1. 初始化本地 Ollama 模型
ollama_llm = ChatOllama(
    model="llama3.1"
)

# 2. 创建 PromptTemplate
template = """你是一位翻译家，请将以下文本翻译为英文：
{text}
"""
prompt =  PromptTemplate.from_template(template)

# 3. Parse the outputparser
parser = StrOutputParser()

# 4. 创建 LLMChain
llm_chain =  prompt | ollama_llm | parser

# 5. 执行一次测试
input_text = "你好，世界！今天天气真好。"
result = llm_chain.invoke({ "text": input_text })
print("翻译结果：", result)
